package com.lagou.threadpool;

import org.apache.dubbo.common.URL;
import org.apache.dubbo.common.threadpool.support.fixed.FixedThreadPool;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;
import java.util.concurrent.*;

public class WatchingThreadPool extends FixedThreadPool implements Runnable {
    private static final Logger LOGGER = LoggerFactory.getLogger(WatchingThreadPool.class);
    //
    private static final double TP90 =0.90;
    private static final double TP99 =0.99;
    private final Map<URL, ThreadPoolExecutor> THREAD_POOLS=new ConcurrentHashMap<>();
    public WatchingThreadPool(){
        //每隔5秒打印线程使用情况
        Executors.newSingleThreadScheduledExecutor().scheduleWithFixedDelay(this,1,5, TimeUnit.SECONDS);
    }
    //通过父类创建线程池
    @Override
    public Executor getExecutor(URL url){
        final Executor executor = super.getExecutor(url);
        if (executor instanceof ThreadPoolExecutor){
            THREAD_POOLS.put(url,(ThreadPoolExecutor)executor);
        }
        return executor;
    }

    @Override
    public void run() {
        //遍历线程池
        for(Map.Entry<URL,ThreadPoolExecutor> entry:THREAD_POOLS.entrySet()){
            final URL url = entry.getKey();
            final ThreadPoolExecutor executor = entry.getValue();
            //计算相关指标
            int activeCount = executor.getActiveCount();
            int corePoolSize = executor.getCorePoolSize();
            double usedPercent =activeCount/(corePoolSize*1.0);
            LOGGER.info("线程池执行状态：[{}/{}:{}%]",activeCount,corePoolSize,usedPercent*100);
            if (usedPercent >TP90){
                LOGGER.warn("达到TP90！host:{} 当前使用率是：{}，URL:{}",url.getIp(),usedPercent*100,url);
            }
            if (usedPercent >TP99){
                LOGGER.warn("达到TP99！host:{} 当前使用率是：{}，URL:{}",url.getIp(),usedPercent*100,url);
            }
        }

    }
}
